package com.bosssoft.trainee.service.imp;

import cn.hutool.core.bean.BeanUtil;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.bosssoft.trainee.common.exception.Asserts;
import com.bosssoft.trainee.dao.RoleDao;
import com.bosssoft.trainee.mbg.dto.RoleDTO;
import com.bosssoft.trainee.mbg.mapper.RoleNapper;
import com.bosssoft.trainee.mbg.po.Resource;
import com.bosssoft.trainee.mbg.po.Role;
import com.bosssoft.trainee.service.ResourceService;
import com.bosssoft.trainee.service.RoleService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.HashSet;
import java.util.List;
import java.util.Set;

/**
 * @Author: wuxin
 * @Date: 2022/6/17
 * @Project rbac
 * @Version 1.0
 * @Description
 */
@Service
public class RoleServiceImp implements RoleService {
    @Autowired
    RoleNapper roleNapper;
    @Autowired
    RoleDao roleDao;
    @Autowired
    ResourceService resourceService;

    /**
     * 非角色列表获取角色
     *
     * @param pageSize 页大小
     * @param pageNum  页数
     * @return 一页数据
     */
    @Override
    public Page<Role> getRoles(int pageSize, int pageNum) {
        Role example = new Role();
        return roleNapper.selectPage(new Page<>(pageNum, pageSize), new QueryWrapper<>(example));
    }

    /**
     * 从角色列表获取用户
     *
     * @param pageSize 页大小
     * @param pageNum  页数
     * @param example  样例
     * @return 一页数据
     */
    @Override
    public Page<Role> getList(int pageSize, int pageNum, RoleDTO example) {
        Role role = new Role();
        BeanUtil.copyProperties(example, role);
        QueryWrapper<Role> roleqw = new QueryWrapper<>(role);
        return roleNapper.selectPage(new Page<>(pageNum, pageSize), roleqw);
    }

    /**
     * 根据id批量获取Role
     *
     * @param roleIds
     * @return
     */
    @Override
    public List<Role> getRoleByIds(List<Integer> roleIds) {
        Set<Integer> ids = new HashSet<>(roleIds);
        return roleNapper.selectBatchIds(ids);
    }

    /**
     * 新增角色
     *
     * @param roleDTO 新增对象
     * @return 受影响的记录数
     */
    @Override
    public int addRole(RoleDTO roleDTO) {
        if (!checkCode(roleDTO.getCode(), roleDTO.getId())) {
            Role role = new Role();
            BeanUtil.copyProperties(roleDTO, role);
            role.setId(null);
            return roleNapper.insert(role);
        } else {
            Asserts.fail("角色代码被占用");
            return 0;
        }
    }

    /**
     * 更新角色
     *
     * @param roleDTO 更新对象
     * @return 受影响的记录数
     */
    @Override
    public int updateRole(RoleDTO roleDTO) {
        if (!checkCode(roleDTO.getCode(), roleDTO.getId())) {
            Role role = new Role();
            BeanUtil.copyProperties(roleDTO, role);
            return roleNapper.updateById(role);
        } else {
            Asserts.fail("角色代码被占用");
            return 0;
        }
    }

    /**
     * 根据id删除角色
     *
     * @param id 删除对象id
     * @return 受影响的记录数
     */
    @Override
    public int deleteById(int id) {
        return roleNapper.deleteById(id);
    }

    /**
     * 批量获取角色能够访问的资源
     *
     * @param roleIds
     * @return
     */
    @Override
    public List<Resource> getResourceByRoleIds(List<Integer> roleIds) {
        Set<Integer> ids = new HashSet<>(roleIds);
        List<Role> roleList = roleNapper.selectBatchIds(ids);
        HashSet<Integer> resourceIds = new HashSet<>();
        roleList.forEach(role -> resourceIds.addAll(role.getResources()));
        return resourceService.getResourceByIds(resourceIds);
    }

    /**
     * 查询角色代码是否被使用
     *
     * @param code 需要查询的代码
     * @return true:被使用，false：未被使用
     */
    @Override
    public boolean checkCode(String code, Integer id) {
        if (id == null) {
            id = 0;
        }
        return roleDao.checkCode(code, id) > 0;
    }
}
